"""
Reproduce: moto sends duplicate HTTP 100 Continue responses.
"""
import socket
import time

from moto.moto_server.threaded_moto_server import ThreadedMotoServer

server = ThreadedMotoServer(port=15555, verbose=False)
server.start()


def raw_request(method, path, body=None, headers=None):
    sock = socket.create_connection(("127.0.0.1", 15555))
    lines = [f"{method} {path} HTTP/1.1", "Host: localhost:15555"]
    for k, v in (headers or {}).items():
        lines.append(f"{k}: {v}")
    if body:
        lines.append(f"Content-Length: {len(body)}")
    lines.append("")
    lines.append("")
    sock.sendall("\r\n".join(lines).encode())
    if body:
        time.sleep(0.1)
        sock.sendall(body)
    sock.settimeout(2)
    chunks = []
    try:
        while True:
            data = sock.recv(4096)
            if not data:
                break
            chunks.append(data)
    except socket.timeout:
        pass
    sock.close()
    return b"".join(chunks)


# Create bucket
raw_request("PUT", "/test-bucket")

# PUT with Expect: 100-continue
resp = raw_request(
    "PUT",
    "/test-bucket/key",
    body=b'{"test": true}',
    headers={"Content-Type": "application/json", "Expect": "100-continue"},
)

server.stop()

status_lines = [
    l for l in resp.decode(errors="replace").split("\r\n") if l.startswith("HTTP/")
]
print("Response status lines:")
for line in status_lines:
    print(f"  {line}")

n_100 = sum(1 for l in status_lines if "100" in l)
assert n_100 <= 1, f"Expected at most 1 '100 Continue', got {n_100}"
print(f"\nPASS: got {n_100} '100 Continue' response(s)")